from ultralytics import YOLO, RTDETR
from ultralytics.models.rtdetr.model import DERTDETR

if __name__ == '__main__':
    # model = YOLO('bdd100k/yolov8-densefpn-tranqkv-athead.yaml')
    # model = YOLO('bdd100k/yolov8-densefpn-tranqkv-athead.yaml')
    # model = YOLO('yoloads.yaml')
    # model = YOLO('bdd100k/yolov8-densefpn-tranqkv-rtdetr.yaml')
    # model = YOLO('yolov8-patchfpn-trancat.yaml')
    model = YOLO('yolov8-patchfpn.yaml')
    # model = DERTDETR('bdd100k/yolov8n-densefpn-tranqkv-rtdetr.yaml')
    model.info()